iT邦幫忙

2023 iThome 鐵人賽

DAY 22
0
AI & Data

深度學習概念和應用(PyTorch)系列 第 22

DAY22 torchvision庫-2

  • 分享至 

  • xImage
  •  

在DAY18的時候有用過,可以對圖片做出一些處理,而這些處理也會影響到最後訓練的結果,
Transforms
網路上薩摩耶的圖片(圖片來源:https://www.dcard.tw/f/pet/p/233192736)

  1. Resize改變大小
size = 100
transform =  transforms.Resize(size)
new_img = transform(img)
new_img

https://ithelp.ithome.com.tw/upload/images/20231007/20163187CQWGXuXrQu.png
上面是原圖,下面是resize之後的,可以看到比例明顯小了不少。
2. CenterCrop從最中間取一定範圍

size = 300
transform = transforms.Compose([
    transforms.CenterCrop(size),
])
img1 = transform(img)
img1

https://ithelp.ithome.com.tw/upload/images/20231007/20163187PKGSB2p9tQ.png
3. Pad增加邊框

padding = 15
transform = transforms.Compose([
    transforms.Resize((100,150)),
    transforms.Pad(padding, fill=(230,200,250),padding_mode="constant"), 
])
img1 = transform(img)
img1

https://ithelp.ithome.com.tw/upload/images/20231007/201631879kWJZozFvr.png
resize之後加上邊框,還可以透過RGB調整顏色
4. RandomHorizontalFlip水平轉向

transform = transforms.Compose([
    transforms.Resize((100,150)),
    transforms.RandomHorizontalFlip(p=0.9),
])

img1 = transform(img)
img1

https://ithelp.ithome.com.tw/upload/images/20231007/201631878rm7JN82Lm.png
5. RandomHorizontalFlip垂直轉向

transform = transforms.Compose([
    transforms.Resize((100,150)),
    transforms.RandomVerticalFlip(p=0.9),
])
img1 = transform(img)
img1

https://ithelp.ithome.com.tw/upload/images/20231007/201631878nM9RmScWt.png


上一篇
DAY21 Torchvision庫1
下一篇
DAY23 mnist實作 - 數據集處理
系列文
深度學習概念和應用(PyTorch)30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言